import os
import random

# 输入文件目录
input_dir = r'C:\Users\ASUS\Desktop\2024年新生赛出题\2024年新生赛出题(非大一组)\F题\F题_input'
os.makedirs(input_dir, exist_ok=True)

# 设置测试用例数量
num_cases = 4
max_n = 1000

# 随机生成 N 的测试用例
test_cases = set()  # 用集合来避免重复的 N 值
while len(test_cases) < num_cases:
    # 生成接近上限的数来增加测试难度
    n_value = random.randint(max_n - 100000, max_n)
    test_cases.add(n_value)

# 保存到输入文件中
for idx, n_value in enumerate(test_cases, start=1):
    with open(os.path.join(input_dir, f"{idx}.in"), "w") as f:
        f.write(f"{n_value}\n")

print("所有测试数据已生成并保存在指定目录中。")
